
from AccessPoint import *
from ItemGoodFaster import *
from ItemGoodMemory import *
from ItemGoodApTurnOn import *
from ItemGoodKillHuman import *
from ItemGoodEvolutive import *

from ItemBadStopper import *
from ItemBadSlower import *
from ItemBadApTurnOff import *

from InfoImages import *


class EngineObjects(object):

	def __init__(self):
		self.initConstants()

	def initConstants(self):
		self.NUM_AP = {}
		self.NUM_AP[0] = [3, 3, 2]
		self.NUM_AP[1] = [2, 2, 2]
		self.NUM_AP[2] = [2, 1, 2, 1]

		self.NUM_HUMAN = {}
		self.NUM_HUMAN[0] = [1, 1, 2]
		self.NUM_HUMAN[1] = [2, 2, 2]
		self.NUM_HUMAN[2] = [2, 2, 3, 4]

		self.NUM_EV_ITEMS = {}
		self.NUM_EV_ITEMS[0] = [2, 3, 3]
		self.NUM_EV_ITEMS[1] = [3, 3, 4]
		self.NUM_EV_ITEMS[2] = [3, 4, 5, 6]

	def setMap(self, screen):
		self.screen = screen

	def startLevel(self, level, scenario):
		i = 0
		while (i < self.NUM_AP[level][scenario]):
			ap = AccessPoint(InfoImages.getImage(InfoImages.AP), i*100, i*100, 200)
			ap.paint(screen)
			i += 1
